#include <ATen/core/ATen_fwd.h>
#include <iostream>
#include <boost/filesystem.hpp>
#include <NumCpp.hpp>
#include <torch/torch.h>
#include <types.h>
using namespace std;
using namespace torch::indexing;

int main(){

    torch::Tensor tensor = torch::tensor({1,2,3});
    cout << tensor.index({1}) << endl;
    // 输出 2
    cout << tensor.index({None}) << endl;
    // 不加索引值，输出所有元素
    // cout << tensor.index({Ellipsis, "..."}) << endl;
    cout << tensor.index({Slice(1, None, 2)}) << endl;
    cout << tensor.index({true, false}) << endl;
    return 0;
}
